# drl_policy_driver.py
import torch

class DRLPolicy:
    def __init__(self, model_path , adj_matrix):
        # self.model = drl_model.Net()
        # self.model.load_state_dict(torch.load(model_path))
        # self.model.eval()
        self.clear_memory()

    def calculate_next_move(self, evader_node, pursuer1_node  , pursuer2_node):
        # return self.model.forward(evader_node, pursuer1_node, pursuer2_node)
        return evader_node, pursuer1_node, pursuer2_node

    def clear_memory(self):
        # self.model.clear_memory()
        pass